from SparkApiV3 import SparkApi
from evn import id

appid = id['appid']     #填写控制台中获取的 APPID 信息
api_secret = id['api_secret']  #填写控制台中获取的 APISecret 信息
api_key = id['api_key']    #填写控制台中获取的 APIKey 信息
domain = "generalv3"   # 大模型版本（不知道可以去看官方文档）
Spark_url = "ws://spark-api.xf-yun.com/v3.1/chat"  # v3.0环境的地址

text =[
        {"role": "system", "content": "你是太阳王子，你是阿悟和空空的好朋友。你擅长用简短的语言回答问题。"},
]

# length = 0

def getText(role,content):
    jsoncon = {}
    jsoncon["role"] = role
    jsoncon["content"] = content
    text.append(jsoncon)
    return text

def getlength(text):
    length = 0
    for content in text:
        temp = content["content"]
        leng = len(temp)
        length += leng
    return length

def checklen(text):
    while (getlength(text) > 8000):
        del text[0]
    return text
    


if __name__ == '__main__':
    text.clear
    while(1):
        Input = input("\n" +"我:")
        question = checklen(getText("user",Input))
        SparkApi.answer =""
        print("星火:",end = "")
        SparkApi.main(appid,api_key,api_secret,Spark_url,domain,question)
        getText("assistant",SparkApi.answer)
        # print(str(text))

